cd ../..

gpus=$1

python -m trainer.ClassAwareTrainer fit \
    --config configs/model/CRNN_contrastive.yaml \
    --config configs/dataset/desed_with_id.yaml \
    --config configs/trainer/default.yaml \
    --config configs/experiment/contextaware.yaml \
    --model.exp_name="2_baselines/classaware/thd0.95/+data_augm/+regular_MT" \
    --data.init_args.batch_sizes=[12,12,12,192,96] \
    --trainer.devices=${gpus} 